import pymysql      #导入pymysql数据库包

#定义了一个get_conn()函数，用于连接数据库，获取连接
def get_conn():
    try:                     # 使用try except来捕获异常，防止程序乱
        db = pymysql.connect(
            host="localhost",
            user="root",
            passwd="Dqh0160108",
            database="homework",
            charset="utf8"
        )                     # 使用pymysql.connect来连接
        return db
    except pymysql.Error:
        print("数据库连接异常")

#定义了一个close_conn()函数用于在程序执行结束后关闭数据库和游标
def close_conn(cursor,db):
    try:
        if cursor:
            cursor.close()
        if db:
            db.close()
    except pymysql.Error as e:
        print('数据库关闭失败' + str(e))
    finally:
        cursor.close()
        db.close()

print(get_conn())


'''
#创建记录管理员信息的用户表
db = get_conn()        # 数据库连接
cur = db.cursor()      # 创建游标
sql1 = "create table user(username  CHAR(20) primary key, userpwd  CHAR(20) not null )"  #创建表的sql语句
sql2 = "drop table if exists user"     #如果数据库中存在的该表，则先删除表，再创建
cur.execute(sql2)
cur.execute(sql1)

'''

